package com.kool.kadmin.common.util;

import com.kool.kadmin.common.constant.ErrorCode;
import com.kool.kadmin.common.exception.AppException;
import lombok.extern.slf4j.Slf4j;

import java.io.*;
import java.util.zip.ZipEntry;
import java.util.zip.ZipOutputStream;

/**
 * @description: 打包工具
 * @author: luyu
 * @create: 2019/02/12 13:43
 */
@Slf4j
public class ZipUtil {
    /**
     * 打包
     *
     * @param desc
     * @param source
     */
    public static void compress(String desc, String source) {

        try {
            ZipOutputStream out = new ZipOutputStream(new FileOutputStream(desc));
            File sourceFile = new File(source);

            compress(out, sourceFile, sourceFile.getName());

            out.close();
        } catch (FileNotFoundException e) {
            log.error( "打包错误");
            throw new AppException(ErrorCode.SYSTEM_ERROR, e, StringUtil.getUUID());
        } catch (IOException e) {
            log.error( "打包错误");
            throw new AppException(ErrorCode.SYSTEM_ERROR, e, StringUtil.getUUID());
        }
    }

    private static void compress(ZipOutputStream out, File sourceFile, String base)
            throws IOException {
        // 如果路径为目录（文件夹）
        if (sourceFile.isDirectory()) {
            // 取出文件夹中的文件（或子文件夹）
            File[] flist = sourceFile.listFiles();

            if (flist.length == 0) {
                // 如果文件夹为空，则只需在目的地zip文件中写入一个目录进入点
                out.putNextEntry(new ZipEntry(base + "/"));
            } else {
                // 如果文件夹不为空，则递归调用compress，文件夹中的每一个文件（或文件夹）进行压缩
                for (int i = 0; i < flist.length; i++) {
                    compress(out, flist[i], base + "/" + flist[i].getName());
                }
            }
        } else {
            // 如果不是目录（文件夹），即为文件，则先写入目录进入点，之后将文件写入zip文件中
            out.putNextEntry(new ZipEntry(base));
            FileInputStream fos = new FileInputStream(sourceFile);
            BufferedInputStream bis = new BufferedInputStream(fos);

            int tag = 0;
            // 将源文件写入到zip文件中
            while ((tag = bis.read()) != -1) {
                out.write(tag);
            }
            bis.close();
            fos.close();
        }
    }
}
